import time
import os



class JT():

    def __init__(self, driver):
        self.driver = driver



    def jietu(self):
    #封装截图方法

        a = "D:/PyCharm/错误代码截图/"    # 添加一个变量 = 图片存放的绝对路径     #斜杠 绝对路径  最后加/  表示路径结束
        if not os.path.exists(a):       # 判断截图存放路径是否存在
            os.makedirs(a)              # 如果不存在就创建一个这样的路径
        sss = a + time.strftime('%Y-%m-%d__%H-%M-%S' ,time.localtime(time.time())) + ".png"  # 添加一个变量 = 图片存放的路径 + 时间命名 + png后缀
                                # 图片名 不能有特殊符号 空格 ：
        self.driver.get_screenshot_as_file(sss)    # 截图
        time.sleep(1)